import torch
class hparams():
    def __init__(self):
        
        self.spk_list = ['mS0002','mS0008','mS0009','fS0663','fS0523','fS0142']
        self.path_train_wavs = 'data/train'
        self.path_test_wavs = 'data/test'
        self.path_catch_feas = 'data_catch'
        self.n_spk = 6
 
        # 特征提取相关
        self.fs = 16000         # 采样率 
        self.frame_period = 5.0 # 帧移
        self.coded_dim = 36     # mepc 特征维度 
        
        
        # 模型训练相关
        self.n_epoch = 300
        
        self.g_lr = 1e-4
        self.d_lr = 1e-4
        self.c_lr = 1e-4
        
        
        self.decay_g = 1e-9
        self.decay_d = 1e-9
        self.decay_c = 1e-9
        
        self.lambda_cycle = 3
        self.lambda_classifier = 2
        self.lambda_identity = 2
        
      
        self.lr_update_step = 1e4
        self.start_decay = 1e5 # 开始进行lr衰减的步数
        
        
        self.path_save = 'save'
        self.save_step = 5000
        
        
        self.path_eval = 'eval'
        
        
        
        
        
        
        